home *** CD-ROM | disk | FTP | other *** search
/ 17 Bit Software 3: The Continuation / 17-Bit_The_Continuation_Disc.iso / amigan / amigan 5 / memwatch / wbc.a < prev    next >
Encoding:
Text File  |  1994-01-27  |  7.2 KB  |  315 lines
  1.     section    text,CODE
  2. * C initial startup procedure under AmigaDOS
  3. * 
  4. * Requirements:
  5.  
  6.     INCLUDE    "exec/types.i"
  7.     INCLUDE "exec/alerts.i"
  8.     INCLUDE "exec/nodes.i"
  9.     INCLUDE "exec/lists.i"
  10.     INCLUDE "exec/ports.i"
  11.     INCLUDE "exec/libraries.i"
  12.     INCLUDE "exec/tasks.i"
  13.     INCLUDE "libraries/dos.i"
  14.     INCLUDE "libraries/dosextens.i"
  15.     INCLUDE "workbench/startup.i"
  16.  
  17. ; some usefull macros:
  18. xlib    macro
  19.     xref    _LVO\1
  20.     endm
  21.  
  22. callsys    macro
  23.     CALLLIB    _LVO\1
  24.     endm
  25.     
  26.  
  27.     xdef    _XCEXIT            * exit(code) is standard way to leave C.
  28.     xdef    _SaveMem
  29.     xdef    _ValidateMem
  30.  
  31.     xref    _LinkerDB        * linker defined base value
  32.  
  33.     xref    _Debug
  34.     xref    __main            * Name of C program to start with.
  35.     xref    _AbsExecBase
  36.     xref    _FindTask
  37.     xref    _DOSBase
  38.  
  39.     xlib    Alert
  40.     xlib    FindTask
  41.     xlib    Forbid
  42.     xlib    GetMsg
  43.     xlib    OpenLibrary
  44.     xlib    CloseLibrary
  45.     xlib    ReplyMsg
  46.     xlib    Wait
  47.     xlib    WaitPort
  48.     xlib    Open
  49.     xlib    Close
  50.     xlib    CurrentDir
  51.  
  52. *** this is the number of LONGWORDS at the start of memory to be checked
  53. MEMSAVESIZE    equ    64
  54. start:
  55.     move.l    d0,dosCmdLen
  56.     move.l    a0,dosCmdBuf
  57.     move.l    a7,d0            ; save old stack ptr
  58.     movem.l    d1-d6/a0-a6,-(a7)
  59.     move.l    d0,a5
  60.     move.l    _AbsExecBase,a6
  61.     move.l    a6,_SysBase
  62.     move.l    a7,__StackPtr        * Save stack ptr
  63.     clr.l    _WBenchMsg
  64.  
  65. ;------ get the address of our task
  66.     suba.l    a1,a1
  67.     callsys    FindTask
  68.     move.l    d0,a4
  69.  
  70. ;------ are we running as a son of Workbench?
  71.     move.l    pr_CurrentDir(A4),_curdir
  72.     tst.l    pr_CLI(A4)
  73.     beq    fromWorkbench
  74.  
  75. ;=======================================================================
  76. ;====== CLI Startup Code ===============================================
  77. ;=======================================================================
  78.  
  79. fromCLI:
  80.     move.l    a5,D0        ; get top of stack
  81.     sub.l    4(a5),D0    ; compute bottom 
  82.     move.l    D0,__base    ; save for stack checking
  83. ;------    attempt to open DOS library:
  84.     bsr    openDOS
  85.  
  86. ;------ find command name:
  87.     move.l    #start,a0
  88.     clr.l    -(sp)
  89.     jsr    _FindTask
  90.     addq.l    #4,sp
  91.     move.l    d0,a0
  92.     move.l    pr_CLI(a0),a0
  93.     add.l   a0,a0        ; bcpl pointer conversion
  94.     add.l   a0,a0
  95.     move.l    cli_CommandName(a0),a1
  96.     add.l   a1,a1        ; bcpl pointer conversion
  97.     add.l   a1,a1
  98.  
  99. ;------    collect parameters:
  100.     move.l    dosCmdLen,d0        ; get command line length
  101.     move.l    a1,__ProgramName
  102.     addq.l    #1,d0            ; allow for space after command    
  103.  
  104.     clr.w    -(A7)            ; set null terminator for command line
  105.     addq.l    #1,D0            ; force to even number of bytes
  106.     andi.w    #$fffe,D0        ;(round up)
  107.     sub.l    D0,A7            ; make room on stack for command line
  108.     subq.l    #2,D0
  109.     clr.w    0(A7,D0)
  110.  
  111. ;------ copy command line onto stack
  112.     move.l    dosCmdLen,d0        ; get command line length
  113.     move.l    dosCmdBuf,a0
  114.     move.l    d0,d2
  115.     subq.l    #1,d0
  116.  
  117. copy_line:
  118.     move.b    0(A0,D0.W),0(A7,D2.W)    ; copy command line to stack
  119.     subq.l    #1,d2
  120.     dbf    d0,copy_line
  121.     move.b    #' ',0(a7,d2.w)        ; add space between command and parms
  122.     subq.l    #1,d2
  123.  
  124. copy_cmd:
  125.     move.l    A7,A1
  126.     move.l    A1,-(A7)        ; push command line address
  127.     bra    goto_main                * call C entrypoint
  128.  
  129. ;=======================================================================
  130. ;====== Workbench Startup Code =========================================
  131. ;=======================================================================
  132.  
  133. fromWorkbench:
  134.  
  135.     move.l    TC_SPLOWER(a4),__base    ; set base of stack
  136. ;------ open the DOS library:
  137.     bsr    openDOS
  138.  
  139. ;------ we are now set up.  wait for a message from our starter
  140.     bsr    waitmsg
  141.     move.l    d0,_WBenchMsg
  142.     move.l    d0,-(SP)
  143. ;
  144.     move.l    d0,a2            ; get first argument
  145.     move.l    sm_ArgList(a2),d0
  146.     beq.s    do_cons
  147.     move.l    _DOSBase,a6
  148.     move.l    d0,a0
  149.     move.l    wa_Lock(a0),d1
  150.     move.l    d1,_curdir
  151.     callsys    CurrentDir
  152. do_cons:
  153.     move.l    sm_ToolWindow(a2),d1    ; get the window argument
  154.     beq.s    do_main
  155.     move.l    #MODE_OLDFILE,d2
  156.     callsys    Open
  157.     move.l    d0,stdin
  158.     beq.s    do_main
  159.     lsl.l    #2,d0
  160.     move.l    d0,a0
  161.     move.l    fh_Type(a0),pr_ConsoleTask(A4)
  162. do_main:
  163.     move.l    _WBenchMsg,a0        ; get address of workbench message
  164.     move.l    sm_ArgList(a0),a0    ; get address of arguments
  165.     move.l    wa_Name(a0),__ProgramName    ; get name of program
  166.     moveq    #0,d0
  167.     move.l    d0,-(A7)
  168.  
  169. goto_main:
  170.     lea    _LinkerDB,a6        ; load base register
  171.     jsr    __main                * call C entrypoint
  172.     moveq.l    #0,d0            ; set successful status
  173.     bra.s    exit2
  174. ;
  175.  
  176. _XCEXIT:
  177.     move.l    4(SP),d0    ; extract return code
  178. exit2:
  179.     move.l    d0,-(a7)
  180.     move.l    _SysBase,a6
  181.     move.l    _DOSBase,a1
  182.     callsys    CloseLibrary        ; close Dos library
  183. done_1b:
  184. ;------ if we ran from CLI, skip workbench cleanup:
  185.     tst.l    _WBenchMsg
  186.     beq.s    exitToDOS
  187.     move.l    _console_dev,d1
  188.     beq.s    done_2
  189.     callsys    Close
  190. done_2:
  191.     move.l    stdin,d1
  192.     beq.s    done_4
  193.     callsys    Close
  194. done_4:
  195.  
  196. ;------ return the startup message to our parent
  197. ;    we forbid so workbench can't UnLoadSeg() us
  198. ;    before we are done:
  199.     move.l    _AbsExecBase,A6
  200.     callsys Forbid
  201.     move.l    _WBenchMsg,a1
  202.     callsys    ReplyMsg
  203.  
  204. ;------ this rts sends us back to DOS:
  205. exitToDOS:
  206.     MOVE.L    (A7)+,D0
  207.     movea.l  __StackPtr,SP        * restore stack ptr
  208.     movem.l    (a7)+,d1-d6/a0-a6
  209.     rts                * and exit
  210.  
  211. ;-----------------------------------------------------------------------
  212. noDOS:
  213.         ALERT    (AG_OpenLib!AO_DOSLib)
  214.         moveq.l    #100,d0
  215.         bra    exit2
  216.  
  217.  
  218. ;-----------------------------------------------------------------------
  219. ; This routine gets the message that workbench will send to us
  220. ; called with task id in A4
  221.  
  222. waitmsg:
  223.         lea    pr_MsgPort(A4),a0      * our process base
  224.         callsys    WaitPort
  225.         lea    pr_MsgPort(A4),a0      * our process base
  226.         callsys GetMsg
  227.         rts
  228.  
  229. ;-----------------------------------------------------------------------
  230. ;  Open the DOS library:
  231.  
  232. openDOS
  233.         lea    DOSName,A1
  234.         moveq.l    #0,D0
  235.         callsys OpenLibrary
  236.         move.l    D0,_DOSBase
  237.         beq    noDOS
  238.         rts
  239.  
  240. *************************************************************************
  241. * Validate the current memory information and put up an alert if it is
  242. * in sad shape
  243. *************************************************************************
  244. _ValidateMem:
  245.     movem.L    A0/A1,-(A7)
  246.     moveq.l    #MEMSAVESIZE,D0
  247.     LEA    0,A0
  248.     LEA    savearea,A1
  249. checkit:
  250.     cmp.l    (A0)+,(A1)+
  251.     dbne    D0,checkit
  252.     addq    #1,D0
  253.     beq    allok
  254. * restore what got scrod
  255. *    first format the message to be printed
  256. *    we get a pointer to the address and the data portion
  257.     move.l    -(A0),D0    get the value that was put in
  258.     move.l    -(A1),(A0)    and replace with what should have been there
  259.     move.l    12(A7),A1    point to where we should save the result
  260.     move.l    D0,(A1)        save the value
  261.     move.l    A0,4(A1)    and the address for the routine to munge
  262.     moveq.l    #1,D0        set a result value to indicate failure
  263. allok:
  264.     movem.l    (a7)+,A0/A1
  265.     rts
  266.  
  267. _SaveMem:
  268.     movem.L    D0/A0/A1,-(A7)
  269.     moveq.l    #MEMSAVESIZE,D0
  270.     LEA    0,A0
  271.     LEA    savearea,A1
  272. savelp:
  273.     move.l    (A0)+,(A1)+
  274.     DBF    D0,savelp
  275.     movem.l    (A7)+,D0/A0/A1
  276.     rts
  277.  
  278.     section    "__MERGED",DATA
  279. ;
  280.     XDEF    _NULL,_SysBase,_LoadAddress,_console_dev,_WBenchMsg
  281.     XDEF    _curdir,__mbase,__mnext,__msize,__tsize
  282.     XDEF    __oserr,__OSERR,__FPERR,__SIGFPE,__ONERR,__ONEXIT,__ONBREAK
  283.     XDEF    __SIGINT,__ECS
  284.     XDEF    __ProgramName,__StackPtr,__base
  285. ;
  286. _NULL        dc.l    0        ;
  287. savearea    ds.l    MEMSAVESIZE+2    ;for save and compare of memory
  288. __base        dc.l    0        ; base of stack
  289. __mbase        dc.l    0        ; base of memory pool
  290. __mnext        dc.l    0        ; next available memory location
  291. __msize        dc.l    0        ; size of memory pool
  292. __tsize        dc.l    0        ; total size?
  293. __oserr        equ    *
  294. __OSERR        dc.l    0
  295. __FPERR        dc.l    0
  296. __SIGFPE    dc.l    0
  297. __SIGINT    dc.l    0
  298. __ONERR        dc.l    0
  299. __ONEXIT    dc.l    0
  300. __ONBREAK    dc.l    0
  301. __ECS        dc.l    0        ; extended character set flag
  302. _curdir        dc.l    0
  303. _console_dev    dc.l    0
  304. _SysBase    dc.l    0
  305. _LoadAddress    dc.l    0            ; program load address
  306. _WBenchMsg    dc.l    0
  307. __StackPtr    dc.l    0
  308. dosCmdLen    dc.l    0
  309. dosCmdBuf    dc.l    0
  310. stdin        dc.l    0
  311. __ProgramName    dc.l    0
  312. DOSName     DOSNAME
  313.  
  314.     END
  315.